package com.df.ez_game;

import cn.hutool.core.util.StrUtil;
import cn.hutool.json.JSONUtil;

import java.io.ByteArrayOutputStream;
import java.util.Base64;
import java.util.zip.DataFormatException;
import java.util.zip.Deflater;
import java.util.zip.Inflater;

/**
 * 数据压缩存储
 * @author Admin
 * @class CompressUtils
 * @date 2024/4/3 11:27
 */
public class CompressUtils {

    /***
     * Json字符串压缩
     * @author: Admin
     * @date: 2024/4/3 11:28
     * @param unzipStr 压缩后的json字符串
     * @return: java.lang.String
     */
    public static String zipStr(String unzipStr) {
        /*
         * 0~9 压缩等级, 低到高
         * BEST_COMPRESSION = 9       最佳压缩级别
         * BEST_SPEED = 1             压缩级别最快的压缩
         * DEFAULT_COMPRESSION = -1   默认压缩级别
         * DEFAULT_STRATEGY = 0       默认压缩策略
         * DEFLATED = 8               压缩算法的压缩方法(目前唯一支持的压缩方法)
         * FILTERED = 1               压缩策略最适用于大部分数值较小且数据分布随机的数据
         * FULL_FLUSH = 3             压缩刷新模式，用于清楚所有待处理的输出并重置拆卸器
         * HUFFMAN_ONLY = 2           仅用于霍夫曼编码的压缩策略
         * NO_COMPRESSION = 0         不压缩的压缩级别
         * NO_FLUSH = 0               用于实现最佳压缩结果的压缩刷新模式
         * SYNC_FLUSH = 2             用于清除所有未决输出的压缩刷新模式；可能会降低某些压缩算法的效率
         */
        // 使用指定的压缩级别创建一个新的压缩器
        Deflater deflater = new Deflater(Deflater.BEST_COMPRESSION);
        // 设置压缩输入数据
        deflater.setInput(unzipStr.getBytes());
        // 当被调用时，表示压缩应该以输入缓冲区的当前内容结束
        deflater.finish();

        final byte[] bytes = new byte[256];
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream(256);

        while (!deflater.finished()) {
            // 压缩输入数据并用压缩数据填充指定的缓冲区
            int length = deflater.deflate(bytes);
            outputStream.write(bytes, 0, length);
        }

        // 关闭压缩器并丢弃任何未处理的输入
        deflater.end();
        return Base64.getEncoder().withoutPadding().encodeToString(outputStream.toByteArray());
//            return outputStream.toString("ISO-8859-1");
    }

    /***
     * 解压缩压缩过的json串
     * @author: Admin
     * @date: 2023/9/19 15:59
     * @param zipStr 压缩过的json串
     * @return: java.lang.String
     */
    public static String unzipStr(String zipStr) {
        if (StrUtil.isEmpty(zipStr)) {
            return "";
        }
        byte[] decode = Base64.getDecoder().decode(zipStr);
//        byte[] decode = zipStr.getBytes(StandardCharsets.ISO_8859_1);
        // 创建一个新的解压缩器
        Inflater inflater = new Inflater();
        // 设置压缩的输入数据
        inflater.setInput(decode);

        final byte[] bytes = new byte[256];
        ByteArrayOutputStream outputStream = new ByteArrayOutputStream(256);
        try {
            // finished(),如果已达到压缩数据流的末尾，则返回true
            while (!inflater.finished()) {
                // 将字节解压到指定的缓冲区中
                int length = inflater.inflate(bytes);
                outputStream.write(bytes, 0 , length);
            }
        } catch (DataFormatException e) {
            e.printStackTrace();
        } finally {
            // 关闭解压缩器并丢弃任何未处理的输入
            inflater.end();
        }
        return JSONUtil.formatJsonStr(outputStream.toString());
    }
}
